jax==0.4.20
matplotlib==3.5.2
numpy==1.24.3
scipy==1.14.1
